cd ../
gpus=$1
# net.normalization=batch (changed in .yaml file)
python sed_main.py --gpus ${gpus}, --exp_name SupOnly --batch_sizes '[0, 12, 12, 0, 0]'